An improved non-negative matrix factorization algorithm based on genetic algorithm

The non-negative matrix factorization (NMF) algorithm is a classical matrix factorization and dimension reduction method in machine learning and data mining. However, in real problems, we always have to run the algorithm for several times and use the best matrix factorization result as the final output because of the random initialization of the matrix factorization. In this paper, we proposed an improved non-negative matrix factorization algorithm based on genetic algorithm (GA), which uses the internal parallelism and the random search of genetic algorithm to get the optimal solution of matrix factorization. It could have larger searching area and higher accuracy in matrix factorization. In the document clustering problem, we use the TDT2 dataset and design several contrast experiments on the classical NMF and the improved NMF based on genetic algorithm, the experiment results show that our improved non-negative matrix factorization algorithm has higher clustering accuracy and better robustness.